package main

import (
	"baidu"
	"encoding/base64"
	"file"
	"fmt"
	"io/ioutil"
	"strconv"
)

const (
	imgSuffix      = ".jpg.png.PNG.JPG"
	outputFileName = "pic2word.txt"
)

func main() {

	// 1.展示标题
	message.ShowTip()

	// 2. 循环遍历当前文件夹的图片信息
	picFileCount, picFiles := file.SearchPic("./", imgSuffix)

	// 打印文案
	fmt.Println("总共" + strconv.Itoa(picFileCount) + "个文件")
	fmt.Println("解析文件中...")

	// 3. 循环调用百度AI接口进行文字转换
	data := make(map[string]interface{})
	for i := 0; i < picFileCount; i++ {
		fmt.Print("第" + strconv.Itoa(i+1) + "/" + strconv.Itoa(picFileCount) + "个文件" + picFiles[i] + "解析中...")
		ff, _ := ioutil.ReadFile(picFiles[i])

		// base64编码
		data["image"] = base64.StdEncoding.EncodeToString(ff)
		// 调用百度AI图片转文字接口
		words := baidu.BasicGeneral(data)

		// 写入文件
		err := file.WriteWithIoutil(outputFileName, words+"\n\n")
		if err == nil {
			fmt.Println("成功！！！")
		} else {
			fmt.Println("失败！！！")
		}
		fmt.Println(words)
	}
}
